package Algorithm.KMP;

import java.util.Arrays;

//KMP算法 匹配字符串
public class KMPAlgorithm {
	public static void main(String[] args) {
		String str1 = "BBC ABCDAB ABCDABCDABDE";
		String str2 = "ABCDABD";

		int[] next = kmpNext("ABCDABD");
		System.out.println("next = " + Arrays.toString(next));
		System.out.println("匹配到的首字符下标" + KMPSearch(str1, str2, next));
	}


	/**
	 * KMP搜索算法
	 *
	 * @param str1 原字符串
	 * @param str2 目标子串
	 * @param next 子串对应的部分匹配表
	 * @return 若为 -1 没有匹配到，否则返回首字符的下标
	 */
	public static int KMPSearch(String str1, String str2, int[] next) {
		int times = 0;
		//遍历 原字符串
		for (int i = 0, j = 0; i < str1.length(); i++) {
			times++;
			//需要考虑，当原字符串的第 i 个和子串的第 j 个不相同时,调整 j 的大小
			while (j > 0 && str1.charAt(i) != str2.charAt(j)) {
				j = next[j - 1];
			}

			//当原字符串的第 i 个和子串的第 j 个相同时
			if (str1.charAt(i) == str2.charAt(j)) {
				j++;
			}
			if (j == str2.length()) {
				//找到
				//因为这里是先执行 j++ 再执行完循环的代码块
				//最后 i++，所以 i 会比 j 小一，所以需要 +1
				System.out.println("匹配次数:" + times);
				return i - j + 1;
			}
		}
		System.out.println("匹配次数:" + times);
		return -1;
	}

	//获取到目标字串的部分匹配值
	public static int[] kmpNext(String dest) {
		//创建一个和目标字符串长度相同的数组
		int[] next = new int[dest.length()];
		//如果字符串为一时 部分匹配值为0
		next[0] = 0;
		for (int i = 1, j = 0; i < dest.length(); i++) {
			//当某两个字符不同时，需要从 next[j-1]获取新的 j
			//直到发现有两个字符相同时退出
			while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
				j = next[j - 1];
			}

			//表示某两个字符相同，部分匹配值 + 1
			if (dest.charAt(i) == dest.charAt(j)) {
				j++;
			}
			next[i] = j;
		}
		return next;
	}


}
